import numpy as np
from openpyxl import Workbook
import glob
import os

# 修改为指定的文件夹路径
folder_path = r'D:\0-兰理\13-土耳其地震\土耳其地震+SSI谱\μ-model\SSI-inelastic-pro\Result-BL\R=2.0\a0=1.0,s=3.0'
flist = glob.glob(os.path.join(folder_path, '*.txt'))  # 获取指定文件夹下所有的txt文件

# 创建一个新的Excel工作簿
workbook = Workbook()
sheet = workbook.active

column_index = 1  # 从第一列开始

for filename in flist:
    data = np.loadtxt(filename, delimiter=None, skiprows=0, usecols=0)  # 读取当前txt文件的数据
    sample1 = np.loadtxt(filename, delimiter=None, skiprows=0, usecols=0, max_rows=1, dtype=float)
    sample = np.reshape(sample1, (-1,))
    array = np.concatenate((sample, data))  # 将第一个元素与数据合并成一个数组

    # 将数据逐行写入Excel表格
    for row_index, value in enumerate(array, start=1):
        sheet.cell(row=row_index, column=column_index, value=value)

    column_index += 1  # 每个txt文件写入一列

# 保存Excel文件到与txt文件相同的文件夹中
output_path = os.path.join(folder_path, f'{os.path.basename(folder_path)}_output.xlsx')

# 保存Excel文件
workbook.save(output_path)

print(f'Excel文件已保存到: {output_path}')
